home *** CD-ROM | disk | FTP | other *** search
/ 17 Bit Software 3: The Continuation / 17-Bit_The_Continuation_Disc.iso / amigan / amigan 5 / secmap / secmap.c < prev    next >
C/C++ Source or Header  |  1994-01-27  |  12KB  |  344 lines
  1. /*  :ts=8 bk=0 
  2.  * Disk mapper.  Uses trackdisk.device to grab and read sector bitmap 
  3.  * to discover what's allocated, then displays a (hopefully) pretty picture 
  4.  * showing disk useage. 
  5.  * 
  6.  * Crufted together by Leo Schwab while desperately bored.  8606.8 
  7.  * Turned into something working for the Manx compiler.  8607.23 
  8.  */ 
  9.  
  10. #include <exec/types.h> 
  11. #include <exec/memory.h> 
  12. #include <intuition/intuition.h> 
  13. #include <libraries/dos.h> 
  14. #include <devices/trackdisk.h> 
  15.  
  16. #define REV             0L 
  17. #define BLOCKSIZE       TD_SECTOR 
  18. #define NUMBLOCKS       (NUMCYLS * NUMHEADS * NUMSECS) 
  19. #define ROOTBLOCK       (NUMBLOCKS / 2) 
  20. #define BITMAPINDEX     79 
  21. #define NUMLONGS        (NUMBLOCKS / 32) 
  22. #define XX              6L 
  23. #define YY              6L 
  24. #define XOFF            30L 
  25. #define YOFF            25L 
  26. #define BRKOVER         91L     /*  11 * YY + YOFF  */ 
  27. #define SEP             6 
  28.  
  29. extern void     *OpenLibrary(), *OpenDevice(), *OpenWindow(), *AllocMem(), 
  30.                 *Lock(), *Info(), *CreatePort(), *CreateExtIO(); 
  31. extern long     TextLength(); 
  32.  
  33.  
  34. struct NewWindow windef = { 
  35.         0, 0, 640, 180, 
  36.         -1, -1, 
  37.         CLOSEWINDOW, 
  38.         WINDOWDRAG | WINDOWDEPTH | WINDOWCLOSE | SMART_REFRESH | ACTIVATE, 
  39.         NULL, NULL, 
  40.         (UBYTE *) "Disk Allocation Map ", 
  41.         NULL, NULL, 0, 0, 0, 0, 
  42.         WBENCHSCREEN 
  43. }; 
  44.  
  45. struct IntuiText ok = { 
  46.         AUTOFRONTPEN, AUTOBACKPEN, 
  47.         AUTODRAWMODE, AUTOLEFTEDGE, AUTOTOPEDGE, 
  48.         AUTOITEXTFONT, 
  49.         (UBYTE *) "OK", 
  50.         AUTONEXTTEXT 
  51. }; 
  52.  
  53. struct IntuiText errmsg = { 
  54.         AUTOFRONTPEN, AUTOBACKPEN, 
  55.         AUTODRAWMODE, AUTOLEFTEDGE, AUTOTOPEDGE, 
  56.         AUTOITEXTFONT, 
  57.         NULL,   /*  Gets filled in later  */ 
  58.         AUTONEXTTEXT 
  59. }; 
  60.  
  61. struct Window   *win; 
  62. struct RastPort *rp; 
  63. struct InfoData *id; 
  64. struct IOExtTD  *diskreq; 
  65. struct MsgPort  *diskport; 
  66. ULONG           diskchangecount, lok, *diskbuffer; 
  67. void            *IntuitionBase, *GfxBase; 
  68.  
  69.  
  70. main (ac, av) 
  71. char *av[]; 
  72. { 
  73.         long k, x, y; 
  74.         int unit, bmsect, free = NUMBLOCKS-2; 
  75.         short i, n, l; 
  76.         char buf[80]; 
  77.  
  78.         if (!ac) {      /*  Run from workbench (ick!)  */ 
  79.                 /*  I'll make this part better someday.....  */ 
  80.                 strcpy (buf, "df0:"); 
  81.         } else {        /*  Run from CLI (thank you)  */ 
  82.                 if (ac == 1) { 
  83.                         printf ("Drive specifier: "); 
  84.                         gets (buf); 
  85.                 } else 
  86.                         strcpy (buf, av[1]); 
  87.         } 
  88.  
  89.         openstuff (); 
  90.  
  91.         if (!(lok = Lock (buf, ACCESS_READ))) 
  92.                 die ("Can't obtain lock for specified device."); 
  93.  
  94.         if (!(id = AllocMem ((long) sizeof (*id), MEMF_CLEAR))) 
  95.                 die ("Can't get InfoData memory."); 
  96.  
  97.         if (!Info (lok, id)) 
  98.                 die ("Call to Info() failed."); 
  99.  
  100.         if (id -> id_DiskType == ID_NO_DISK_PRESENT) 
  101.                 die ("No disk in drive."); 
  102.  
  103.         unit = id -> id_UnitNumber; 
  104.         FreeMem (id, (long) sizeof (*id));  id = NULL; 
  105.         opendisk (unit); 
  106.         MotorOn (); 
  107.         GetSector ((long) ROOTBLOCK); 
  108.         bmsect = diskbuffer[BITMAPINDEX]; 
  109.         GetSector ((long) bmsect); 
  110.         MotorOff (); 
  111.  
  112.         /* 
  113.          * At this point, we now have the bitmap in the disk buffer. 
  114.          * This is what the first longword in the bitmap represents: 
  115.          * 
  116.          *      333322222222221111111111 
  117.          *      32109876543210987654321098765432        Sector # 
  118.          *      -------------------------------- 
  119.          *      10100101001010010010010001010011        Longword 
  120.          *      MSB                          LSB 
  121.          * 
  122.          * If the bit is set, the sector is free. 
  123.          * 
  124.          * Sectors 0 and 1 contain the boot block.  Thus, the DOS is not 
  125.          * allowed to walk on them, and they are permanently allocated. 
  126.          * However, the bitmap does not reflect this i.e. the bitmap starts 
  127.          * at sector 2, which is indicated by bit 0, longword 1.  This means 
  128.          * we have to offset everything by two.  Can you say "crock?" 
  129.          * 
  130.          * Actually, I suppose this is good, since it prevents some would-be 
  131.          * smart person from un-allocating those sectors and messing things 
  132.          * up badly. 
  133.          * 
  134.          * Bitmap starts at longword 1; longword 0 appears to be some kind 
  135.          * of significant garbage (checksum?) 
  136.          * 
  137.          * For more info on how I found the bitmap sector, consult the DOS 
  138.          * Technical Reference Manual, p. 1-1 - 1-2 
  139.          */ 
  140.  
  141.         SetAPen (rp, 3L); 
  142.         RectFill (rp, XOFF, YOFF, XX+XOFF, YY+YY+YOFF); /*  Show first two  */ 
  143.         for (i=1; i<=NUMLONGS; i++) { 
  144.                 k = diskbuffer[i]; 
  145.                 for (n=0; n<32; n++) 
  146.                         /*  Bits progress from low to high order  */ 
  147.                         if (i<NUMLONGS || n<30) {       /*  Ignore last two  */ 
  148.                                 if (~k & 1) { 
  149.                                         /*  Perform icky conversion  */ 
  150.                                         free--; 
  151.                                         l = (i-1 << 5) + n + 2; 
  152.                                         x = (l / 22) * XX + XOFF; 
  153.                                         y = (l % 22) * YY + YOFF; 
  154.                                         if (y >= BRKOVER) 
  155.                                                 y += SEP; 
  156.                                         RectFill (rp, x, y, x+XX-1, y+YY-1); 
  157.                                 } 
  158.                                 k >>= 1; 
  159.                         } 
  160.         } 
  161.  
  162.         /*  Draw grid so we can see  */ 
  163.         SetAPen (rp, 1L); 
  164.         for (x=XOFF; x<=80*XX+XOFF; x += XX) { 
  165.                 Move (rp, x, YOFF); 
  166.                 Draw (rp, x, YOFF+11*YY); 
  167.                 Move (rp, x, BRKOVER+SEP); 
  168.                 Draw (rp, x, BRKOVER+SEP+11*YY); 
  169.         } 
  170.         for (y=0; y<=11*YY; y += YY) { 
  171.                 Move (rp, XOFF, y+YOFF); 
  172.                 Draw (rp, XOFF+80*XX, y+YOFF); 
  173.                 Move (rp, XOFF, y+SEP+BRKOVER); 
  174.                 Draw (rp, XOFF+80*XX, y+SEP+BRKOVER); 
  175.         } 
  176.  
  177.         /*  Draw map markings  */ 
  178.         Move (rp, XOFF+XX/2, YOFF);       Draw (rp, XOFF+XX/2, YOFF-3); 
  179.         Move (rp, XOFF+80*XX-XX/2, YOFF); Draw (rp, XOFF+80*XX-XX/2, YOFF-3); 
  180.         Move (rp, XOFF, YOFF+YY/2);       Draw (rp, XOFF-3, YOFF+YY/2); 
  181.         Move (rp, XOFF, YOFF+YY*10+YY/2); Draw (rp, XOFF-3, YOFF+YY*10+YY/2); 
  182.         Move (rp, XOFF-1, YOFF-3);        Text (rp, "0", 1L); 
  183.         Move (rp, XOFF+79*XX-1, YOFF-3);  Text (rp, "79", 2L); 
  184.         Move (rp, XOFF-12, YOFF+6);       Text (rp, "0", 1L); 
  185.         Move (rp, XOFF-20, YOFF+11*YY);   Text (rp, "10", 2L); 
  186.  
  187.         /*  Do labels  */ 
  188.         sprintf (buf, "Bitmap on sector %d", bmsect); 
  189.         Move (rp, XOFF, 176L); 
  190.         Text (rp, buf, (long) strlen (buf)); 
  191.         sprintf (buf, "Sectors free: %d", free); 
  192.         Move (rp, 250L, 176L); 
  193.         Text (rp, buf, (long) strlen (buf)); 
  194.         sprintf (buf, "Allocated: %d", (int) NUMBLOCKS-free); 
  195.         Move (rp, 450L, 176L); 
  196.         Text (rp, buf, (long) strlen (buf)); 
  197.         Move (rp, 520L, 60L); 
  198.         Text (rp, "Surface 0", 9L); 
  199.         Move (rp, 520L, 132L); 
  200.         Text (rp, "Surface 1", 9L); 
  201.  
  202.         /*  Wait for close gadget  */ 
  203.         Wait (1L << win -> UserPort -> mp_SigBit); 
  204.         closestuff (); 
  205. } 
  206.  
  207.  
  208. openstuff () 
  209. { 
  210.         if (!(IntuitionBase = OpenLibrary ("intuition.library", REV))) { 
  211.                 /* 
  212.                  * If we can't open Intuition, then we can't use 
  213.                  * AutoRequest () 
  214.                  */ 
  215.                 printf ("Intuition failed; you'll have to use logic.\n"); 
  216.                 closestuff (); 
  217.                 exit (100); 
  218.         } 
  219.  
  220.         if (!(GfxBase = OpenLibrary ("graphics.library", REV))) { 
  221.                 printf ("Art shop closed.\n"); 
  222.                 closestuff (); 
  223.                 exit (100); 
  224.         } 
  225.  
  226.         if (!(win = OpenWindow (&windef))) { 
  227.                 printf ("Window painted shut.\n"); 
  228.                 closestuff (); 
  229.                 exit (100); 
  230.         } 
  231.         rp = win -> RPort; 
  232. } 
  233.  
  234. opendisk (unit) 
  235. int unit; 
  236. { 
  237.         long err; 
  238.         char *buf[80]; 
  239.  
  240.         if (!(diskport = CreatePort (NULL, NULL))) 
  241.                 die ("No port."); 
  242.  
  243.         if (!(diskreq = CreateExtIO (diskport, (long) sizeof (*diskreq)))) 
  244.                 die ("Can't make IO block."); 
  245.  
  246.         if (err = OpenDevice (TD_NAME, (long) unit, diskreq, NULL)) { 
  247.                 sprintf (buf, "Can't get at disk; err = %ld.", err); 
  248.                 DeleteExtIO (diskreq, (long) sizeof (*diskreq)); 
  249.                 diskreq = NULL; 
  250.                 die (buf); 
  251.         } 
  252.  
  253.         if (!(diskbuffer = AllocMem (BLOCKSIZE, MEMF_CLEAR | MEMF_CHIP))) 
  254.                 die ("Can't allocate disk buffer."); 
  255.  
  256.         diskreq -> iotd_Req.io_Command = TD_CHANGENUM; 
  257.         DoIO (diskreq); 
  258.         diskchangecount = diskreq -> iotd_Req.io_Actual; 
  259. } 
  260.  
  261. closestuff () 
  262. { 
  263.         if (lok) 
  264.                 UnLock (lok); 
  265.         if (diskreq) { 
  266.                 CloseDevice (diskreq); 
  267.                 DeleteExtIO (diskreq, (long) sizeof (*diskreq)); 
  268.         } 
  269.  
  270.         if (diskbuffer) 
  271.                 FreeMem (diskbuffer, (long) BLOCKSIZE); 
  272.         if (id) 
  273.                 FreeMem (id, (long) sizeof (*id)); 
  274.         if (diskport) 
  275.                 DeletePort (diskport); 
  276.         if (win) 
  277.                 CloseWindow (win); 
  278.         if (GfxBase) 
  279.                 CloseLibrary (GfxBase); 
  280.         if (IntuitionBase) 
  281.                 CloseLibrary (IntuitionBase); 
  282. } 
  283.  
  284. die (str) 
  285. UBYTE *str; 
  286. { 
  287.         errmsg.IText = str; 
  288.         AutoRequest (win, &errmsg, NULL, &ok, NULL, NULL, 
  289.                      TextLength (rp, str, (long) strlen (str)) + 40, 46L); 
  290.         closestuff (); 
  291.         exit (100); 
  292. } 
  293.  
  294. /* 
  295.  * The following routines were stolen from an RKM example by Bob Peck and 
  296.  * hacked up a bit. 
  297.  */ 
  298.  
  299. GetSector (sector) 
  300. long sector; 
  301. { 
  302.         LONG offset = sector * BLOCKSIZE; 
  303.  
  304.         diskreq -> iotd_Req.io_Length = BLOCKSIZE;       
  305.         diskreq -> iotd_Req.io_Data = (APTR) diskbuffer;  
  306.                 /* show where to put the data when read */ 
  307.         diskreq -> iotd_Req.io_Command = ETD_READ; 
  308.                 /* check that disk not changed before reading */ 
  309.         diskreq -> iotd_Count = diskchangecount; 
  310.          
  311.         /* convert from cylinder, head, sector to byte-offset value to get 
  312.          * right one (as dos and everyone else sees it)...*/ 
  313.          
  314.         /* driver reads one CYLINDER at a time (head does not move for 
  315.          * 22 sequential sector reads, or better-put, head doesnt move for 
  316.          * 2 sequential full track reads.) 
  317.          */ 
  318.          
  319.         diskreq -> iotd_Req.io_Offset = offset; 
  320.         DoIO (diskreq); 
  321.         return (0); 
  322. } 
  323.  
  324. MotorOn()
  325. { 
  326.         /* TURN ON DISK MOTOR ... old motor state is returned in io_Actual */ 
  327.         diskreq -> iotd_Req.io_Length = 1; 
  328.         /* this says motor is to be turned on */ 
  329.         diskreq -> iotd_Req.io_Command = TD_MOTOR; 
  330.         /* do something with the motor */ 
  331.         DoIO (diskreq); 
  332.         return (0); 
  333. } 
  334.  
  335. MotorOff() 
  336. { 
  337.         diskreq -> iotd_Req.io_Length = 0;       
  338.         /* says that motor is to be turned on */ 
  339.         diskreq -> iotd_Req.io_Command = TD_MOTOR;       
  340.         /* do something with the motor */ 
  341.         DoIO (diskreq); 
  342.         return (0); 
  343. } 
  344.